package com.dyz.baseAlgorithm.hashTable;

import java.util.Scanner;

/**
 * @author: daiyizheng
 * @date: 2021/5/17 12:53
 * @description:
 */
public class HashTabDemo {
    public static void main(String[] args) {
        // 创建哈希表
        HashTable hashTab = new HashTable(7);

        // 写一个简单的菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while(true) {
            System.out.println("add:  添加雇员");
            System.out.println("list: 显示雇员");
            System.out.println("find: 查找雇员");
            System.out.println("exit: 退出系统");
            System.out.println();
            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    // 创建 雇员
                    EmpNode empNode = new EmpNode(id, name);
                    hashTab.add(empNode);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "find":
                    System.out.println("请输入要查找的id");
                    id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                default:
                    break;
            }
        }
    }
}

//创建HashTab 管理多条链表
class HashTable{
    private EmpLinkedList[] empLinkedListArray;
    private int size; // 表示有多少条链表

    //构造器
    public HashTable(int size){
        this.size = size;
        // 初始化empLinkedListArray
        empLinkedListArray = new EmpLinkedList[size];
        for(int i=0;i<size; i++){
            empLinkedListArray[i] = new EmpLinkedList();
        }
    }

    //添加雇员
    public void add(EmpNode empNode){
        //根据员工的id，得到该员工应当添加到哪条链表
        int empLinkedListNo = hashFun(empNode.id);
        //将emp添加到对应的链表中
        empLinkedListArray[empLinkedListNo].add(empNode);
    }

    //遍历雇员
    public void list(){
        for(int i=0; i<size; i++){
            empLinkedListArray[i].list(i);
        }
    }

    //根据输入的id ，查找雇员
    public  void findEmpById(int id){
        // 使用散列函数确定到哪条链表查找
        int empLinkedListNO = hashFun(id);
        EmpNode empNode = empLinkedListArray[empLinkedListNO].findEmpById(id);
        if (empNode != null) {// 找到
            System.out.printf("在第%d条链表中找到 雇员 id = %d\n", (empLinkedListNO + 1), id);
        } else {
            System.out.println("在哈希表中，没有找到该雇员~");
        }
    }

    //编写散列函数，使用一个简单取模法
    public int hashFun(int id){
        return id % size;
    }
}

//创建雇员
class EmpNode{
    public int id;
    public String name;
    public EmpNode next;//next 默认为null

    public  EmpNode(int id, String name){
        super();
        this.id = id;
        this.name = name;
    }
}

//创建EmpLinkedlist, 表示链表
class EmpLinkedList{
    //首指针，指向第一个EmpNode,因此我们这个链表的head，是直接指向第一个EmpNode
    private EmpNode head;
    //添加雇员到链表
    //说明
    //假定，当添加雇员时，id是自增长，即id的分配总是小到大
    //因此我们将该雇员直接加入到本链表的最后即可
    public void add(EmpNode empNode){
        //如果是第一个雇员
        if(head == null){
            head = empNode;
        }
        //如果不是第一个雇员，则使用一个辅助的指针，帮助定位到最后
        EmpNode curEmp = head;
        while (true){
            if(curEmp.next==null){ // 说明到链表最后
                break;  // 后移
            }
            // 退出时直接将emp 加入链表
            curEmp.next = empNode;
        }
    }

    //遍历链表的雇员信息
    public void list(int no){
        if(head == null){// 说明链表为空
            System.out.println("第 " + (no + 1) + " 链表为空");
        }
        System.out.print("第 " + (no + 1) + " 链表的信息为");
        EmpNode curEmp = head; // 辅助指针
        while (true){
            System.out.printf(" => id=%d name=%s ", curEmp.id, curEmp.name);
            if (curEmp.next == null){
                break;
            }
            curEmp = curEmp.next;// 后移，遍历
        }
        System.out.println();
    }

    //根据id查找雇员
    //如果查找到，就返回emp，如果没有找到，就返回null
    public EmpNode findEmpById(int id){
        //判断链表是否为空
        if(head == null){
            System.out.println("链表为空");
            return null;
        }
        // 辅助指针
        EmpNode curEmp = head;
        while (true) {
            if (curEmp.id == id) {// 找到
                break;// 这时curEmp就指向要查找的雇员
            }
            curEmp = curEmp.next;// 后移
            // 退出
            if (curEmp == null) {// 说明遍历当前链表没有找到该雇员
                break;
            }
        }
        return curEmp;
    }



}